rf <- arx_forecaster(
epi_data = jhu,
outcome = "death_rate",
predictors = c("case_rate", "death_rate", "fb-survey"),
trainer = parsnip::rand_forest(mode = "regression"), # use ranger
args_list = arx_args_list(
ahead = 14, # 2-week horizon
lags = list(c(0:4, 7, 14), c(0, 7, 14), c(0:7, 14)), # bunch of lags
levels = c(0.01, 0.025, 1:19/20, 0.975, 0.99), # 23 ForecastHub quantiles
quantile_by_key = "geo_value" # vary q-forecasts by location
)
)